package com.liang.leetcode.string;

/**
 * 541. 反转字符串 II
 * 给定一个字符串 s 和一个整数 k，从字符串开头算起，每计数至 2k 个字符，就反转这 2k 字符中的前 k 个字符。
 *  如果剩余字符少于 k 个，则将剩余字符全部反转。
 *  如果剩余字符小于 2k 但大于或等于 k 个，则反转前 k 个字符，其余字符保持原样。
 * 示例 1：
 *  输入：s = "abcdefg", k = 2
 *  输出："bacdfeg"
 * 示例 2：
 *  输入：s = "abcd", k = 2
 *  输出："bacd"
 * 示例 3：
 *  输入：s = "abcdefgh", k = 3
 *  输出："cbadefhg"
 *  
 * 理解：每隔 2k 个字符反转前 k 个，末尾不够 2k 个时，翻转前 k 个，不够 k 个时全部反转
 * 关键在于找到字符翻转的起始和结束下标
 */
public class String02_ReverseStr {
    public static void main(String[] args) {
        String str = "abcdefgh";
        int k = 3;
        String reserveStr = reverseStr(str, k);
        System.out.println(reserveStr);
    }

    /**
     * 解法一：
     * 测试用例：s = "abcdefgh", k = 3;
     */
    public static String reverseStr(String s, int k) {
        char[] chars = s.toCharArray();
        // 每 2k 个字符作为一次循环
        for (int i = 0; i < chars.length; i += 2 * k) {
            // 翻转 2k 中的前 k 个字符
            if (i + k <= chars.length) {
                reverseString(chars, i, i + k - 1);
                continue;
            }
            // 剩余字符小于 2k 个，全部翻转
            reverseString(chars, i, chars.length - 1);
        }
        return new String(chars);
    }

    /**
     * 解法二：
     * 测试用例：s = "abcdefgh", k = 3;
     */
    public static String reverseStr2(String s, int k) {
        char[] chars = s.toCharArray();
        // 每 2k 个字符作为一次循环
        for (int i = 0; i < chars.length; i += 2 * k) {
            // 先判断尾数够不够 k 个来取决 j 指针的下标
            int j = Math.min(chars.length - 1, i + k - 1);
            reverseString(chars, i, j);
        }
        return new String(chars);
    }

    /**
     * 翻转字符串函数
     */
    public static void reverseString(char[] s, int i, int j) {
        while (i <= j) {
            char temp = s[i];
            s[i] = s[j];
            s[j] = temp;
            i++;
            j--;
        }
    }
}
